package leets.leet2364;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

class Solution {
    public long countBadPairs(int[] nums) {
        long res=0;
        int n= nums.length;
        for (int i = 0; i < nums.length; i++) {
            nums[i]=i-nums[i];
        }

        long inver=0;
        Map<Integer,Long> map=new HashMap<>();
        for (int num : nums) {
           map.put(num,map.getOrDefault(num,0L)+1);
        }
        for (Map.Entry<Integer,Long> entry: map.entrySet()){
            long v=entry.getValue();
            inver+=(v*(v-1))/2;
        }
        res= (long) n *(n-1)/2-inver;
        return res;
    }
}
